package github.kawaiior.yuanchuxiuzhen.util;

import net.minecraft.world.biome.Biome;

import java.util.HashSet;
import java.util.Set;

public class BiomeUtil {

    static final Set<Biome.Category> MAIN_WORLD_BIOME = new HashSet<>();

    static {
        MAIN_WORLD_BIOME.add(Biome.Category.TAIGA);
        MAIN_WORLD_BIOME.add(Biome.Category.EXTREME_HILLS);
        MAIN_WORLD_BIOME.add(Biome.Category.JUNGLE);
        MAIN_WORLD_BIOME.add(Biome.Category.MESA);
        MAIN_WORLD_BIOME.add(Biome.Category.PLAINS);
        MAIN_WORLD_BIOME.add(Biome.Category.SAVANNA);
        MAIN_WORLD_BIOME.add(Biome.Category.ICY);
        MAIN_WORLD_BIOME.add(Biome.Category.BEACH);
        MAIN_WORLD_BIOME.add(Biome.Category.FOREST);
        MAIN_WORLD_BIOME.add(Biome.Category.OCEAN);
        MAIN_WORLD_BIOME.add(Biome.Category.DESERT);
        MAIN_WORLD_BIOME.add(Biome.Category.RIVER);
        MAIN_WORLD_BIOME.add(Biome.Category.SWAMP);
        MAIN_WORLD_BIOME.add(Biome.Category.MUSHROOM);
    }

    public static boolean isMainWorldBiome(Biome.Category category){
        return MAIN_WORLD_BIOME.contains(category);
    }

}
